#pragma once

#include <iostream>
#include <regex>
#include <sstream>
#include <iomanip>
#include <fstream>
#include <string>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <atomic>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    class StringUtil
    {
    public:
        // 切分字符串
        // str: 输入型，目标要切分的字符串
        // target: 输出型，保存切分完毕的结果
        // sep: 指定的分割符
        static void SplitString(const std::string &str, std::vector<std::string> *target, const std::string &sep)
        {
            // boost split
            boost::split((*target), str, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };

    class JsonUtil
    {
    public:
        // 转换为UTF8形式进行打印
        static std::string decodeUnicodeEscape(const std::string &str)
        {
            std::string result = str;
            std::regex unicode_regex("\\\\u([0-9a-fA-F]{4})");

            std::sregex_iterator it(result.begin(), result.end(), unicode_regex);
            std::sregex_iterator end;

            while (it != end)
            {
                std::smatch match = *it;
                std::string hex_str = match[1].str();

                // 将16进制字符串转换为Unicode码点
                int code_point = std::stoi(hex_str, nullptr, 16);

                // 将Unicode码点转换为UTF-8
                std::string utf8_char;
                if (code_point <= 0x7F)
                {
                    utf8_char += static_cast<char>(code_point);
                }
                else if (code_point <= 0x7FF)
                {
                    utf8_char += static_cast<char>(0xC0 | (code_point >> 6));
                    utf8_char += static_cast<char>(0x80 | (code_point & 0x3F));
                }
                else
                {
                    utf8_char += static_cast<char>(0xE0 | (code_point >> 12));
                    utf8_char += static_cast<char>(0x80 | ((code_point >> 6) & 0x3F));
                    utf8_char += static_cast<char>(0x80 | (code_point & 0x3F));
                }

                result.replace(match.position(), match.length(), utf8_char);
                it = std::sregex_iterator(result.begin(), result.end(), unicode_regex);
            }

            return result;
        }
    };

    class TimeUtil
    {
    public:
        // 获得毫秒时间戳
        static std::string GetTimeMs()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec * 1000 + _time.tv_usec / 1000);
        }
    };

    class FileUtil
    {
    public:
        // 判断文件是否存在
        static bool IsFileExists(const std::string &path_name)
        {
            // stat 可以获得文件的全部属性
            struct stat st;
            if (stat(path_name.c_str(), &st) == 0)
            {
                // 获取属性成功，文件已经存在
                return true;
            }

            return false;
        }

        // 生成唯一文件名
        static std::string UniqFileName()
        {
            // 原子性递增
            static std::atomic_uint id(0); // C++11自带的一个原子值
            id++;

            // 毫秒级时间戳
            std::string ms = TimeUtil::GetTimeMs();
            std::string uniq_id = std::to_string(id);

            return ms + "_" + uniq_id;
        }

        // 向target中写入内容
        static bool WriteFile(const std::string &target, const std::string &content)
        {
            std::ofstream out(target);
            if (!out.is_open())
            {
                return false;
            }
            out.write(content.c_str(), content.size());
            out.close();
            return true;
        }

        // 从target中读取内容到 content, keep表示是否要保存换行符
        static bool ReadFile(const std::string &target, std::string *content, bool keep = false)
        {
            // 清空content内容
            (*content).clear();

            std::ifstream in(target);
            if (!in.is_open())
            {
                return false;
            }
            std::string line;
            // getline:不保存行分割符,有些时候需要保留\n,
            // getline内部重载了强制类型转化,可以直接在while处使用
            while (std::getline(in, line))
            {
                (*content) += line;
                (*content) += (keep ? "\n" : "");
            }
            in.close();
            return true;
        }
    };

    const std::string temp_path = "./temp/"; // 临时文件所放置的位置

    class PathUtil
    {
    public:
        // 为 file_name 添加一个后缀
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }

        // 编译时需要有临时文件
        // 构建源文件路径+后缀的完整文件名 1234 -> ./temp/1234.cpp
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }

        // 构建可执行程序的完整路径+后缀名
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }

        // 构建编译错误文档的完整路径+后缀名
        static std::string CompilerError(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compile_error");
        }

        // 运行时需要的临时文件
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }

        // 标准输出
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }

        // 标准错误
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };
};